Hackerrank Arithmetic Operators solution in python